package org.zjvis.datascience.common.util;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import java.util.List;
import org.zjvis.datascience.common.dto.KeyValueDTO;

public class KeyValueUtil {
    public static JSONArray toLong(List<KeyValueDTO> kvs, String keyName, String valueName){
        JSONArray ret = new JSONArray();
        for (KeyValueDTO kv: kvs){
            JSONObject jsonObject = new JSONObject();
            jsonObject.put(keyName, kv.getK());
            jsonObject.put(valueName, Long.valueOf(kv.getV()));
            ret.add(jsonObject);
        }
        return ret;
    }

    public static JSONArray toInt(List<KeyValueDTO> kvs, String keyName, String valueName){
        JSONArray ret = new JSONArray();
        for (KeyValueDTO kv: kvs){
            JSONObject jsonObject = new JSONObject();
            jsonObject.put(keyName, kv.getK());
            jsonObject.put(valueName, Integer.valueOf(kv.getV()));
            ret.add(jsonObject);
        }
        return ret;
    }
}
